import os

BASE_DIR = os.path.dirname(os.path.abspath(__file__))

DATA_DIR = os.path.join(BASE_DIR, "data")
RESULT_DIR = os.path.join(DATA_DIR, "results")
EXCEL_DIR = os.path.join(DATA_DIR, "excels")
IMAGE_DIR = os.path.join(DATA_DIR, "images")

for dir_path in [DATA_DIR, RESULT_DIR, EXCEL_DIR, IMAGE_DIR]:
    if not os.path.exists(dir_path):
        os.makedirs(dir_path)

TRAIN_RESULT_FILE = os.path.join(RESULT_DIR, "train_result.txt")
PREDICT_RESULT_FILE = os.path.join(RESULT_DIR, "predict_result.txt")
PREPROCESS_FILE = os.path.join(RESULT_DIR, "preprocess_result.txt")
MODEL_INFO_FILE = os.path.join(RESULT_DIR, "model_info.txt")
EVOLUTION_RULES_FILE = os.path.join(RESULT_DIR, "evolution_rules.txt")
ANALYSIS_EXCEL_FILE = os.path.join(EXCEL_DIR, "result_analysis.xlsx")

VISUALIZATION_IMAGES = [
    os.path.join(IMAGE_DIR, "viz_1.png"),
    os.path.join(IMAGE_DIR, "viz_2.png"),
    os.path.join(IMAGE_DIR, "viz_3.png")
]
HISTORICAL_IMAGES = [
    os.path.join(IMAGE_DIR, "history_1.png"),
    os.path.join(IMAGE_DIR, "history_2.png")
]
COMPARISON_IMAGES = [
    os.path.join(IMAGE_DIR, "compare_1.png"),
    os.path.join(IMAGE_DIR, "compare_2.png"),
    os.path.join(IMAGE_DIR, "compare_3.png")
]